Macos install cpu-only pytorch

2018-10-26

Macos without NVIDIA GPU, we need to use the CPU-only pytorch.

Install pytorch

1	conda install pytorch torchvision -c pytorch

1	pip install torch torchvision

Got RuntimeError:

RuntimeError: Attempting to deserialize object on a 
CUDA device but torch.cuda.is_available() is False. If 
you are running on a CPU-only machine, please use 
torch.load with map_location='cpu' to map your storages 
to the CPU.

modify your python code

torch.load(model_weights)
||
||
vv
torch.load(model_weights, map_location=torch.device('cpu'))